package com.ruoyi.controller;

import com.ruoyi.service.AnythingLLMService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;import reactor.core.publisher.Mono;

import javax.annotation.Resource;
import java.util.Map;

/**
 * AIChatController 类用于处理与 AI 聊天相关的 HTTP 请求。
 * 它通过 AnythingLLMService 提供的 service 方法来响应客户端的聊天请求。
 */
@RestController
@RequestMapping("/api/ai")
public class AIChatController {
   @Resource
    // llmService 用于处理与 AI 聊天相关的逻辑。
    private final AnythingLLMService llmService;

    /**
     * 构造函数注入 AnythingLLMService。
     *
     * @param llmService AnythingLLMService 的实例，用于处理聊天请求。
     */
    @Autowired
    public AIChatController(AnythingLLMService llmService) {
        this.llmService = llmService;
    }

    /**
     * 处理聊天请求的端点。
     * 它接受一个包含 "message" 字段的请求体，将该消息传递给 AI 处理，
     * 并返回 AI 的响应。
     *
     * @param request 包含 "message" 字段的请求体，用于向 AI 发送消息。
     * @return AI 的响应消息，以 JSON 格式返回。
     */
    @PostMapping("/chat")
    public ResponseEntity<String> handleChatRequest(@RequestBody Map<String, String> request) {
        try {
            // 从请求中提取用户消息。
            String message = request.get("message");
            // 使用 llmService 聊天并获取响应。
            String response = llmService.chatWithDocument(message);
            // 返回成功的 HTTP 响应，包含 AI 的消息。
            return ResponseEntity.ok()
                    .contentType(MediaType.APPLICATION_JSON_UTF8)
                    .body(response);
        } catch (Exception e) {
            // 如果发生异常，返回内部服务器错误的 HTTP 响应，并包含错误信息。
            return ResponseEntity.status(HttpStatus.INTERNAL_SERVER_ERROR)
                    .body("{\"error\": \"" + e.getMessage() + "\"}");
        }
    }
}
